<?php
/************************************************************************
 Omnitrix - PHP Development Framework
 Copyright (C) 2012-2018 amonest(eastson@outlook.com)

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
************************************************************************/

class omnitrix_html_message extends omnitrix_html_container
{
	protected $_class;

	public function __construct($class = null)
	{
		if (null == $class) {
			$class = 'success';
		}
		$this->set_class($class);
	}

	public function set_class($class)
	{
		$this->_class = $class;
		return $this;
	}

	public function render_html()
	{
		$html = otx_html_tag_start('div', array(
				'class' => $this->_class
		));
		$html .= parent::render_html();
		$html .= otx_html_tag_end('div');
		return $html;
	}
}